<?php
/**
 * Zend Framework
 *
 * LICENSE
 *
 * This source file is subject to the new BSD license that is bundled
 * with this package in the file LICENSE.txt.
 * It is also available through the world-wide-web at this URL:
 * http://framework.zend.com/license/new-bsd
 * If you did not receive a copy of the license and are unable to
 * obtain it through the world-wide-web, please send an email
 * to license@zend.com so we can send you a copy immediately.
 *
 * @category   Zend
 * @package    Zend_Controller
 * @copyright  Copyright (c) 2005-2011 Zend Technologies USA Inc. (http://www.zend.com)
 * @license    http://framework.zend.com/license/new-bsd     New BSD License
 * @version    $Id: HttpTestCase.php 23775 2011-03-01 17:25:24Z ralph $
 */

/**
 * @see Zend_Controller_Response_Http
 */
require_once 'Zend/Controller/Response/Http.php';

/**
 * Zend_Controller_Response_HttpTestCase
 *
 * @uses Zend_Controller_Response_Http
 * @package Zend_Controller
 * @subpackage Response
 */
class Zend_Controller_Response_HttpTestCase extends Zend_Controller_Response_Http {
	/**
	 * "send" headers by returning array of all headers that would be sent
	 *
	 * @return array
	 */
	public function sendHeaders() {
		$headers = array ();
		foreach ( $this->_headersRaw as $header ) {
			$headers [] = $header;
		}
		foreach ( $this->_headers as $header ) {
			$name = $header ['name'];
			$key = strtolower ( $name );
			if (array_key_exists ( $name, $headers )) {
				if ($header ['replace']) {
					$headers [$key] = $header ['name'] . ': ' . $header ['value'];
				}
			} else {
				$headers [$key] = $header ['name'] . ': ' . $header ['value'];
			}
		}
		return $headers;
	}
	
	/**
	 * Can we send headers?
	 *
	 * @param  bool $throw
	 * @return void
	 */
	public function canSendHeaders($throw = false) {
		return true;
	}
	
	/**
	 * Return the concatenated body segments
	 *
	 * @return string
	 */
	public function outputBody() {
		$fullContent = '';
		foreach ( $this->_body as $content ) {
			$fullContent .= $content;
		}
		return $fullContent;
	}
	
	/**
	 * Get body and/or body segments
	 *
	 * @param  bool|string $spec
	 * @return string|array|null
	 */
	public function getBody($spec = false) {
		if (false === $spec) {
			return $this->outputBody ();
		} elseif (true === $spec) {
			return $this->_body;
		} elseif (is_string ( $spec ) && isset ( $this->_body [$spec] )) {
			return $this->_body [$spec];
		}
		
		return null;
	}
	
	/**
	 * "send" Response
	 *
	 * Concats all response headers, and then final body (separated by two
	 * newlines)
	 *
	 * @return string
	 */
	public function sendResponse() {
		$headers = $this->sendHeaders ();
		$content = implode ( "\n", $headers ) . "\n\n";
		
		if ($this->isException () && $this->renderExceptions ()) {
			$exceptions = '';
			foreach ( $this->getException () as $e ) {
				$exceptions .= $e->__toString () . "\n";
			}
			$content .= $exceptions;
		} else {
			$content .= $this->outputBody ();
		}
		
		return $content;
	}
}
